Forgot to add this to handle code types