Added tiny if statement