Fix mistake

Fixes compile errors