Remove hard coded problem type