Fixing if statement last commit not properly tested