Forgot to fix strategy class