Code error tweaks