Fixed import path in models