Test looks much nicer now