Fixing import for Forge

Still need to work on the other forge related things