fix compile error

Forgot to close a problem environment