Super bad but working tests