Fix pretty simple tests