Add FK restriction on test