Make into proper test