fixed swapped import

worked before because swapped imports were both being imported as base
classes