Fix issue where code block wasn't focused on creation