Dependency issue broke code. Fixed