Add if statement to make it an enum if no block