diff --git a/src/common/alignment.h b/src/common/alignment.h
index 3379a6967535b1be4817263b16e780181140caa2..617b14d9b7746db5a21aa4e4a463fab788f721be 100644
--- a/src/common/alignment.h
+++ b/src/common/alignment.h
@@ -20,8 +20,9 @@ constexpr T AlignDown(T value, std::size_t size) {
 }
 
 template <typename T>
-constexpr T AlignBits(T value, T align) {
-    return (value + ((1 << align) - 1)) >> align << align;
+constexpr T AlignBits(T value, std::size_t align) {
+    static_assert(std::is_unsigned_v<T>, "T must be an unsigned value.");
+    return static_cast<T>((value + ((1ULL << align) - 1)) >> align << align);
 }
 
 template <typename T>