CON-907 | Fix some more more errors