Minor code improvement - removed if statement