Better test fixing yeah buddy