Add proper length check