Write proper tests