Fixes small code smell